#pragma once
#include <string>
#include "Image.h"
#include "Structure.h"

class LeNet
{
private:
	const DL::Tensor4D<float>* conv1Weight;
	const DL::Tensor1D<float>* conv1Bias;
	const DL::Tensor4D<float>* conv2Weight;
	const DL::Tensor1D<float>* conv2Bias;
	const DL::Tensor2D<float>* linear1Weight;
	const DL::Tensor1D<float>* linear1Bias;
	const DL::Tensor2D<float>* linear2Weight;
	const DL::Tensor1D<float>* linear2Bias;
	const DL::Tensor2D<float>* linear3Weight;
	const DL::Tensor1D<float>* linear3Bias;
public:
	LeNet(std::string paramsFile);
	~LeNet();
	int GetResultIndex(Image& image);
	std::string GetResultLable(Image& image);
private:

};